package com.jpro.framework.ip.core.utils;

import cn.hutool.core.io.resource.ResourceUtil;
import cn.hutool.core.net.NetUtil;
import com.jpro.framework.ip.core.Area;
import com.jpro.framework.ip.core.enums.AreaTypeEnum;
import lombok.SneakyThrows;
import lombok.extern.slf4j.Slf4j;
import org.lionsoul.ip2region.xdb.Searcher;

import java.io.IOException;

import static com.jpro.framework.ip.core.Area.ID_CHINA;

/**
 * IP 工具类
 * <p>
 * IP 数据源来自 ip2region.xdb 精简版，基于 <a href="https://gitee.com/lionsoul/ip2region"/> 项目
 *
 * @author jprocms
 */
@Slf4j
public class IPUtils {

    /**
     * 初始化 SEARCHER
     */
    @SuppressWarnings("InstantiationOfUtilityClass")
    private final static IPUtils INSTANCE = new IPUtils();

    /**
     * IP 查询器，启动加载到内存中
     */
    private static Searcher SEARCHER;

    /**
     * 私有化构造
     */
    private IPUtils() {
        try {
            long now = System.currentTimeMillis();
            byte[] bytes = ResourceUtil.readBytes("ip2region.xdb");
            SEARCHER = Searcher.newWithBuffer(bytes);
            log.info("启动加载 IPUtils 成功，耗时 ({}) 毫秒", System.currentTimeMillis() - now);
        } catch (IOException e) {
            log.error("启动加载 IPUtils 失败", e);
        }
    }

    /**
     * 查询 IP 对应的地区编号
     *
     * @param ip IP 地址，格式为 127.0.0.1
     * @return 地区id
     */
    @SneakyThrows
    public static Integer getAreaId(String ip) {
        //内网ip或者是本机ipv6
        if (NetUtil.localIpv6s().contains(ip)||NetUtil.isInnerIP(ip)) {
            return ID_CHINA;
        }
        return Integer.parseInt(SEARCHER.search(ip.trim()));
    }

    /**
     * 查询 IP 对应的地区编号
     *
     * @param ip IP 地址的时间戳，格式参考{@link Searcher#checkIP(String)} 的返回
     * @return 地区编号
     */
    @SneakyThrows
    public static Integer getAreaId(long ip) {
        return Integer.parseInt(SEARCHER.search(ip));
    }

    /**
     * 查询 IP 对应的地区
     *
     * @param ip IP 地址，格式为 127.0.0.1
     * @return 地区
     */
    public static Area getArea(String ip) {
        return AreaUtils.getArea(getAreaId(ip));
    }

    /**
     * 查询 IP 对应的地区
     *
     * @param ip IP 地址的时间戳，格式参考{@link Searcher#checkIP(String)} 的返回
     * @return 地区
     */
    public static Area getArea(long ip) {
        return AreaUtils.getArea(getAreaId(ip));
    }

    public static String getCountry(Area area){
        if(area!=null){
            if(AreaTypeEnum.CITY.getType().equals(area.getType())){
                return area.getParent().getParent().getName();
            }else if(AreaTypeEnum.PROVINCE.getType().equals(area.getType())){
                return area.getParent().getName();
            }else if(AreaTypeEnum.COUNTRY.getType().equals(area.getType())){
                return area.getName();
            }else if(AreaTypeEnum.DISTRICT.getType().equals(area.getType())){
                return area.getParent().getParent().getParent().getName();
            }
        }
        return "未知";
    }

    public static String getProvice(Area area){
        if(area!=null){
            if(AreaTypeEnum.CITY.getType().equals(area.getType())){
                return area.getParent().getName();
            } else if (AreaTypeEnum.PROVINCE.getType().equals(area.getType())) {
                return area.getName();
            } else if (AreaTypeEnum.DISTRICT.getType().equals(area.getType())) {
                return area.getParent().getParent().getName();
            }
        }
        return "未知";
    }

    public static Area getCity(Area area) {
        if (area != null) {
            if (AreaTypeEnum.CITY.getType().equals(area.getType()) || AreaTypeEnum.PROVINCE.getType().equals(area.getType())) {
                return area;
            } else if (AreaTypeEnum.DISTRICT.getType().equals(area.getType())) {
                return area.getParent();
            }
        }
        return area;
    }

    public static void main(String[] args) {
        final Area area = getArea("183.216.110.8");
        System.out.println(getCountry(area));
        System.out.println(getProvice(area));
        System.out.println(getCity(area));
    }


}
